import os

from dotenv import load_dotenv

load_dotenv()

LLM_KEY_LOCAL = os.getenv('LLM_KEY_LOCAL')
LLM_KEY_HUOSHAN = os.getenv('LLM_KEY_HUOSHAN')
LLM_KEY_BAILIAN = os.getenv('LLM_KEY_BAILIAN')
LLM_KEY_DEEPSEEK = os.getenv('LLM_KEY_DEEPSEEK')

default_ai_configs = {
    "local": {
        "model": "deepseek-r1-distill-qwen-7b",
        "url": "http://127.0.0.1:1234/v1/chat/completions",
        "key": LLM_KEY_LOCAL
    },
    "local_embedding": {
      "model": "text-embedding-nomic-embed-text-v1.5",
      "url": "http://127.0.0.1:1234/v1/embeddings",
      "key": LLM_KEY_LOCAL
    },
    "huoshan-embedding-240715": {
      "model": 'doubao-embedding-text-240715',
      "url": 'https://ark.cn-beijing.volces.com/api/v3/embeddings',
      "key": LLM_KEY_HUOSHAN,
    },
    'deepseek-v3': {
      "model": 'deepseek-chat',
      "url": 'https://api.deepseek.com/chat/completions',
      "key": LLM_KEY_DEEPSEEK,
    },
    'deepseek-r1': {
      "model": 'deepseek-reasoner',
      "url": 'https://api.deepseek.com/chat/completions',
      "key": LLM_KEY_DEEPSEEK,
    },
    'huoshan-deepseek-v3': {
      "model": 'deepseek-v3-250324',
      "url": 'https://ark.cn-beijing.volces.com/api/v3/chat/completions',
      "key": LLM_KEY_HUOSHAN,
    },
    'huoshan-deepseek-r1': {
      "model": 'deepseek-r1-distill-qwen-7b-250120',
      "url": 'https://ark.cn-beijing.volces.com/api/v3/chat/completions',
      "key": LLM_KEY_HUOSHAN,
    },
    'huoshan-doubao': {
      "model": 'doubao-1-5-lite-32k-250115',
      "url": 'https://ark.cn-beijing.volces.com/api/v3/chat/completions',
      "key": LLM_KEY_HUOSHAN,
    },
    'bailian-qwen-turbo': {
      "model": 'qwen-turbo',
      "url": 'https://dashscope.aliyuncs.com/compatible-mode/v1/chat/completions',
      "key": LLM_KEY_BAILIAN,
    }
}
